import os
import shutil
import glob
def split(all_dir,test_dir,num):

    files = os.listdir(all_dir)
    files.sort()

    if os.path.exists(test_dir):
        shutil.rmtree(test_dir)
        os.makedirs(test_dir)
    else:
        os.makedirs(test_dir)

    for i in range(num):
        tmp=len(files)-1-i
        print(files[tmp])
        shutil.move(os.path.join(all_dir,files[tmp]),os.path.join(test_dir,files[tmp]))
if __name__=="__main__":
    split("E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-image_result_pre_reg",
          "E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-image_result_pre_reg_test",5)
    split("E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-image_result_pre_reg",
          "E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-image_result_pre_reg_test", 5)
    split("E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-label_result_pre_reg",
          "E:\MIA_CODE_DATA\zhuang_data\MMWHS\MRI\mr-label_result_pre_reg_test", 5)
    split("E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-label_result_pre_reg",
          "E:\MIA_CODE_DATA\zhuang_data\MMWHS\CT\\train\ct-label_result_pre_reg_test", 5)